from langchain_openai import ChatOpenAI
import os

# Description: Configuration file
class Config:
    def __init__(self):
        """
        Initializes the configuration for the agent
        """
        self.BASE_LLM = ChatOpenAI(model="gpt-4", temperature=0.2, base_url=os.getenv('PATSNAP_API_URL'), default_headers={"X-Ai-Engine": os.getenv('OPENAI_MODEL_HEADERS')}, api_key=os.getenv('PATSNAP_API_KEY'))
        self.FACTUAL_LLM = ChatOpenAI(model="gpt-4o-mini", temperature=0.0, base_url=os.getenv('PATSNAP_API_URL'), default_headers={"X-Ai-Engine": os.getenv('OPENAI_MODEL_HEADERS')}, api_key=os.getenv('PATSNAP_API_KEY'))
        self.DEBUG = True